"""
Proxy registration for FastMCP
"""

from fastmcp import FastMCP
from .proxy_manager import ProxyManager


class ProxyRegistry:
    """Registry for proxy management."""

    def __init__(self, mcp: FastMCP):
        self.mcp = mcp
        self.proxy_manager = ProxyManager(mcp)

    async def register(self):
        """Register proxy management and start enabled proxies."""
        # Start all enabled proxies
        await self.proxy_manager.start_enabled_proxies()

    def get_manager(self) -> ProxyManager:
        """Get the proxy manager instance."""
        return self.proxy_manager
